#include "qhomeeditor.h"
#include <QtGui/QApplication>

#include <QtCore/QTimer>
#include <QtGui/QApplication>
#include <QtGui/QtGui>
#include <QtCore/QtCore>

#include <osgViewer/CompositeViewer>
#include <osgViewer/ViewerEventHandlers>

#include <osgGA/TrackballManipulator>

#include <osgDB/ReadFile>

#include <osgQt/GraphicsWindowQt>

#include <iostream>

class ViewerWidget : public QWidget, public osgViewer::CompositeViewer
{
public:
	ViewerWidget(osgViewer::ViewerBase::ThreadingModel threadingModel=osgViewer::CompositeViewer::SingleThreaded) : QWidget()
	{
		setThreadingModel(threadingModel);

		QWidget* widget1 = addViewWidget( createCamera(0,0,100,100), osgDB::readNodeFile("cow.osgt") );
		QWidget* widget2 = addViewWidget( createCamera(0,0,100,100), osgDB::readNodeFile("glider.osgt") );
		QWidget* widget3 = addViewWidget( createCamera(0,0,100,100), osgDB::readNodeFile("axes.osgt") );
		QWidget* widget4 = addViewWidget( createCamera(0,0,100,100), osgDB::readNodeFile("fountain.osgt") );

		QGridLayout* grid = new QGridLayout;
		grid->addWidget( widget1, 0, 0 );
		grid->addWidget( widget2, 0, 1 );
		grid->addWidget( widget3, 1, 0 );
		grid->addWidget( widget4, 1, 1 );
		setLayout( grid );

		connect( &_timer, SIGNAL(timeout()), this, SLOT(update()) );
		_timer.start( 10 );
	}

	QWidget* addViewWidget( osg::Camera* camera, osg::Node* scene )
	{
		osgViewer::View* view = new osgViewer::View;
		view->setCamera( camera );
		addView( view );

		view->setSceneData( scene );
		view->addEventHandler( new osgViewer::StatsHandler );
		view->setCameraManipulator( new osgGA::TrackballManipulator );

		osgQt::GraphicsWindowQt* gw = dynamic_cast<osgQt::GraphicsWindowQt*>( camera->getGraphicsContext() );
		return gw ? gw->getGLWidget() : NULL;
	}

	osg::Camera* createCamera( int x, int y, int w, int h, const std::string& name="", bool windowDecoration=false )
	{
		osg::DisplaySettings* ds = osg::DisplaySettings::instance().get();
		osg::ref_ptr<osg::GraphicsContext::Traits> traits = new osg::GraphicsContext::Traits;
		traits->windowName = name;
		traits->windowDecoration = windowDecoration;
		traits->x = x;
		traits->y = y;
		traits->width = w;
		traits->height = h;
		traits->doubleBuffer = true;
		traits->alpha = ds->getMinimumNumAlphaBits();
		traits->stencil = ds->getMinimumNumStencilBits();
		traits->sampleBuffers = ds->getMultiSamples();
		traits->samples = ds->getNumMultiSamples();

		osg::ref_ptr<osg::Camera> camera = new osg::Camera;
		camera->setGraphicsContext( new osgQt::GraphicsWindowQt(traits.get()) );

		camera->setClearColor( osg::Vec4(0.2, 0.2, 0.6, 1.0) );
		camera->setViewport( new osg::Viewport(0, 0, traits->width, traits->height) );
		camera->setProjectionMatrixAsPerspective(
			30.0f, static_cast<double>(traits->width)/static_cast<double>(traits->height), 1.0f, 10000.0f );
		return camera.release();
	}

	virtual void paintEvent( QPaintEvent* event )
	{ 
		frame(); 
		QWidget::paintEvent(event);
	}

protected:

	QTimer _timer;
};

int main(int argc, char *argv[])
{
	osg::ArgumentParser arguments(&argc, argv);

	osgViewer::ViewerBase::ThreadingModel threadingModel = osgViewer::ViewerBase::CullDrawThreadPerContext;
	while (arguments.read("--SingleThreaded")) threadingModel = osgViewer::ViewerBase::SingleThreaded;
	while (arguments.read("--CullDrawThreadPerContext")) threadingModel = osgViewer::ViewerBase::CullDrawThreadPerContext;
	while (arguments.read("--DrawThreadPerContext")) threadingModel = osgViewer::ViewerBase::DrawThreadPerContext;
	while (arguments.read("--CullThreadPerCameraDrawThreadPerContext")) threadingModel = osgViewer::ViewerBase::CullThreadPerCameraDrawThreadPerContext;

	QApplication a(argc, argv);
	QHomeEditor w;
	ViewerWidget* viewWidget = new ViewerWidget(threadingModel);

	w.setCentralWidget(viewWidget);
	w.show();
	return a.exec();
}
